package com.zhn;

import java.util.*;
import java.util.stream.IntStream;
import java.util.stream.Stream;

/**
 * 给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的 下一个更大元素 。
 *
 * 数字 x 的 下一个更大的元素 是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1
 *
 *
 * 示例 1:
 * 输入: nums = [1,2,1]
 * 输出: [2,-1,2]
 * 解释: 第一个 1 的下一个更大的数是 2；
 * 数字 2 找不到下一个更大的数；
 * 第二个 1 的下一个最大的数需要循环搜索，结果也是 2
 */
public class NextGreaterElements {
    public static void main(String[] args) {
        int[] arr = {1,2,3,4,5};
        nextGreaterElements(arr);
    }
    //拼接方法会有重复计算，导致重复的入栈出栈操作
    public static int[] nextGreaterElements(int[] nums) {
        int n = nums.length;
        int[] res = new int[n];
        Arrays.fill(res,-1);
        Stack<Integer> stack = new Stack<Integer>();
        for(int i=0;i<n;i++){
            while(!stack.isEmpty() && nums[i] > nums[stack.peek()]){
                res[stack.pop()] = nums[i];
            }
            stack.push(i);
        }
        for(int i=0;i<n;i++){
            while(!stack.isEmpty() && nums[i] > nums[stack.peek()]){
                res[stack.pop()] = nums[i];
            }
            // 这里不需要再加入栈操作，否则和拼接无异
        }
        return res;
    }
}
